# syntax=docker/dockerfile:1
FROM pytorch/pytorch:1.9.1-cuda11.1-cudnn8-runtime
WORKDIR /yolov3
COPY . .
RUN pip3 install -r requirements.txt -i https://pypi.tuna.tsinghua.edu.cn/simple
CMD ["python3", "main.py"]
